package com.aterrizar.filtro;

import java.util.ArrayList;
import java.util.List;

import com.aterrizar.asiento.Asiento;

public class FiltroClase implements Filtro {

	private List<String> listClases;

	public FiltroClase(List<String> list) {
		setListClases(list);
	}

	public List<Asiento> filtrar(List<Asiento> asientos) {
		List<Asiento> asientosFiltrados = new ArrayList<Asiento>();
		for (Asiento asiento : asientos) {
			String claseAsiento = asiento.getClase();
			if (listClases.contains(claseAsiento)) {
				asientosFiltrados.add(asiento);
			}
		}
		return asientosFiltrados;
	}

	public void setListClases(List<String> listClases) {
		this.listClases = listClases;
	}

	public List<String> getListClases() {
		return listClases;
	}

}
